package com.xxx.server.config;

import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.xxx.server.pojo.MailConstants;
import com.xxx.server.pojo.MailLog;
import com.xxx.server.service.IMailLogService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 *
 *    RabbitMQ配置类
 *
 * @ClassName RabbitMQconfig
 * @Descriotion TODO
 * @Author zyh
 * @Date 2021/4/25 7:23
 * @Version 1.0
 **/
@Configuration
public class RabbitMQconfig {

    private static final Logger LOGGER = LoggerFactory.getLogger(RabbitMQconfig.class);

    @Autowired
    private CachingConnectionFactory cachingConnectionFactory;

    @Autowired
    private IMailLogService mailLogService;

    //消息、；回调
    @Bean
    public RabbitTemplate rabbitTemplate(){

        RabbitTemplate rabbitTemplate = new RabbitTemplate(cachingConnectionFactory);

        /*
        * 消息确认回调，确认消息是否到达broker
        * data：消息唯一标识
        * ack：确认结果
        * cause：失败原因
        * */
        rabbitTemplate.setConfirmCallback((data, ack, cause) -> {
            String msgId = data.getId();
            if (ack){
                LOGGER.info("{}======>消息发送成功",msgId);
                mailLogService.update(new UpdateWrapper<MailLog>().set("status",1).eq("msgId",msgId));
            }else {
                LOGGER.error("{}======>消息发送失败",msgId);

            }
        });

        /*
        * 消息失败回调，比如router不到queue时回调
        * msg：消息主题
        * repCode：响应码
        * repText：相应描述
        * exchange：交换机
        * routingkey：路由键
        * */
        rabbitTemplate.setReturnCallback((msg,repCode,repText,exchange,routingkey) ->{

            LOGGER.error("{}==========>消息发送queue时失效",msg.getBody());
        });

        return rabbitTemplate;

    }


    //队列
    @Bean
    public Queue queue(){
        return new Queue(MailConstants.MAIL_QUEUE_NAME);

    }

    //交换机
    @Bean
    public DirectExchange directExchange(){
        return new DirectExchange(MailConstants.MAIL_EXCHANGE_NAME);

    }

    //路由键
    @Bean
    public Binding binding(){
        return BindingBuilder.bind(queue()).to(directExchange()).with(MailConstants.MAIL_ROUTEING_KEY_NAME);
    }
}
